package com.example.leetcode.c101_c200.c106;
/**
 * 根据一棵树的中序遍历与后序遍历构造二叉树。
 *
 * 注意:
 * 你可以假设树中没有重复的元素。
 *
 * 例如，给出
 *
 * 中序遍历 inorder = [9,3,15,20,7]
 * 后序遍历 postorder = [9,15,7,20,3]
 * 返回如下的二叉树：
 *
 *     3
 *    / \
 *   9  20
 *     /  \
 *    15   7
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/construct-binary-tree-from-inorder-and-postorder-traversal
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */

import com.example.leetcode.common.TreeNode;
import com.example.leetcode.common.TreeOperate;

import java.util.Arrays;

/**
 * 从中序与后序遍历序列构造二叉树
 * @author zhanpengguo
 * @date 2020-09-25 8:23
 */
public class Solution {

    // 后序遍历的最后一位是根节点
    // 找出根节点，在中序遍历中根节点左边是左子树，右边是右子树
    // 确定左子树个数，在后序遍历中再确定左子树的根节点；右子树同理
    // 依次类推

    public TreeNode buildTree(int[] inorder, int[] postorder) {
        if (inorder.length == 0) {
            return null;
        }
        int rootVal = postorder[postorder.length - 1];
        int index = 0;
        for (int i = 0; i < inorder.length; i++) {
            if (inorder[i] == rootVal) {
                index = i;
                break;
            }
        }
        int[] inLeft = Arrays.copyOfRange(inorder, 0, index);
        int[] postLeft = Arrays.copyOfRange(postorder, 0, index);
        int[] inRight = Arrays.copyOfRange(inorder, index + 1, inorder.length);
        int[] postRight = Arrays.copyOfRange(postorder, index, postorder.length - 1);
        TreeNode root = new TreeNode(rootVal);
        root.left = buildTree(inLeft, postLeft);
        root.right = buildTree(inRight, postRight);
        return root;
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        int[] inorder = {1,2};
        int[] postorder = {2,1};
        TreeNode treeNode = solution.buildTree(inorder, postorder);
        TreeOperate.aftOrder(treeNode);
    }
}
